package com.project.redis;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.parser.ParserConfig;
import com.alibaba.fastjson.serializer.SerializerFeature;
import org.springframework.data.redis.serializer.RedisSerializer;
import org.springframework.data.redis.serializer.SerializationException;

import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;

/**
 * FastJsonRedisSerializer是一个Redis使用FastJson序列化的工具类，主要包括两个方法，分别是序列化和反序列化。
 */
public class FastJsonRedisSerializer<T> implements RedisSerializer<T> {

    /**
     * 配置默认编码方式为UTF-8。
     */
    public static final Charset DEFAULT_CHARSET = StandardCharsets.UTF_8;

    /**
     * clazz 是 FastJsonRedisSerializer 的泛型 T 所对应的Class对象，这里通过构造函数传入。
     */
    private final Class<T> clazz;

    /**
     * 提供了一个有参构造函数，用于创建FastJsonRedisSerializer对象时传入泛型T所对应的Class对象。
     */
    public FastJsonRedisSerializer(Class<T> clazz) {
        super();
        this.clazz = clazz;
    }


    /*
     * 开启 fastJson 自动序列化
     */
    static {
        //        ParserConfig.getGlobalInstance().setAutoTypeSupport(true);  -> 不推荐
        ParserConfig config = new ParserConfig();
        // 添加白名单 org.springframework.security.core.authority.SimpleGrantedAuthority
        config.addAccept("com.project.security.LoginUserInfo");
        config.addAccept("org.springframework.security.core.authority.SimpleGrantedAuthority");
        // 添加黑名单，禁止自动类型转换的类
//        config.addDeny("com.example.model.Admin");

        // 设置自定义的 ParserConfig
        ParserConfig.global = config;
    }


    /**
     * 此方法实现了对象的序列化，将一个对象转成字节数组并返回。如果对象为空，则返回空字节数组。
     * 序列化时，使用了 FastJson 的 toJSONString() 方法将对象转成JSON字符串，同时添加了
     * WriteClassName 选项来使得序列化后的 JSON 字符串包含对象的类名信息。
     */
    @Override
    public byte[] serialize(T t) throws SerializationException {
        if (t == null) {
            return new byte[0];
        }
        return JSON.toJSONString(t, SerializerFeature.WriteClassName).getBytes(DEFAULT_CHARSET);
    }

    /**
     * 此方法实现了对象的反序列化，将一个字节数组转成对象并返回。如果字节数组为空，则返回null。
     * 反序列化时，使用了FastJson的parseObject()方法将JSON字符串转成对象，clazz表示将JSON字符串
     * 转成哪种类型的对象。
     */
    @Override
    public T deserialize(byte[] bytes) throws SerializationException {
        if (bytes == null || bytes.length <= 0) {
            return null;
        }
        String str = new String(bytes, DEFAULT_CHARSET);

        return JSON.parseObject(str, clazz);
    }
}